package com.antoni.dao;

import com.antoni.pojo.Department;
import com.antoni.pojo.Employee;
import org.springframework.stereotype.Repository;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

/**
 * @Author Zhang Chao
 * @Date 2021/5/14 16:31
 * @Version 1.0
 */
@Repository
public class EmployeeDao {

    //模拟数据库中的数据
    private static Map<Integer, Employee> employees = null;
    //员工有所属的部门
    private DepartmentDao departmentDao;

    static{
        employees = new HashMap<>();
        employees.put(1001,new Employee(1001,"AA","A123456@qq.com",1,new Department(101,"教学部")));
        employees.put(1002,new Employee(1002,"BB","B123456@qq.com",0,new Department(102,"市场部")));
        employees.put(1003,new Employee(1003,"CC","C123456@qq.com",1,new Department(103,"教研部")));
        employees.put(1004,new Employee(1004,"DD","D123456@qq.com",0,new Department(104,"运营部")));
        employees.put(1005,new Employee(1005,"EE","E123456@qq.com",1,new Department(105,"后勤部")));
    }

    //主键自增
    private static Integer initId = 1006;

    /**
     * 添加一个员工
     * @param employee
     */
    public void save(Employee employee){
        if(employee.getId()==null){
            employee.setId(initId++);
        }

        employee.setDepartment(departmentDao.getDepartmentById(employee.getDepartment().getId()));

        employees.put(employee.getId(),employee);
    }

    /**
     *
     * @return 查询全部员工信息
     */
    public Collection<Employee> getAll(){
        return employees.values();
    }

    /**
     *
     * @param id
     * @return 通过id查询员工
     */
    public Employee getEmployeeById(Integer id){
        return employees.get(id);
    }

    /**
     * 通过id删除员工信息
     * @param id 员工id
     *
     */
    public void delete(Integer id){
        employees.remove(id);
    }



}
